# -*- coding: utf-8 -*-
# !/usr/bin/env python
"""
-------------------------------------------------
   File Name：     setup
   Description :   
   Author :       lth
   date：          2023/1/13
-------------------------------------------------
   Change Activity:
                   2023/1/13 13:23: create this script
-------------------------------------------------
used to install the static ops
"""
__author__ = 'lth'


from setuptools import setup
from torch.utils import cpp_extension


# setup(name="upfirdn2d",
#       ext_modules=[cpp_extension.CUDAExtension("upfirdn2d",["upfirdn2d.cpp","upfirdn2d_kernel.cu"])],
#       cmdclass={"build_ext": cpp_extension.BuildExtension.with_options(use_ninja=False)}
#       )

setup(name="fused",
      ext_modules=[cpp_extension.CUDAExtension("fused",["fused_bias_act.cpp","fused_bias_act_kernel.cu"])],
      cmdclass={"build_ext": cpp_extension.BuildExtension.with_options(use_ninja=False)}
      )